package com.xiaotu.storage;

import com.xiaotu.common.util.Constants;
import com.xiaotu.common.util.DataRedisKeyUtils;
import com.xiaotu.common.util.SepratorUtil;
import com.xiaotu.storage.converter.AConverter;
import com.xiaotu.storage.converter.ConverterFactory;
import com.xiaotu.storage.model.CacheDataModel;
import net.sf.json.JSONObject;
import org.springframework.stereotype.Component;

import java.util.Date;

/**
 * 网站平台 播放量，热度，豆瓣评分 转储
 *
 * @author zhangxiao
 * @create 2021-01-22 11:04
 */

@Component("TVProcessDataStorageProcessor")
public class TVProcessDataStorageProcessor extends AHSetStorageProcessor {

    private static final String[] FIELD_ARR = new String[]{"id", "tvid", "siteName", "totalPlayCount", "heat", "dbScore", "collectTime"};

    private static final String[] CONVERT_ARR = new String[]{AConverter.STRING_CONVERT,
            AConverter.STRING_CONVERT, AConverter.STRING_CONVERT, AConverter.NUMBER_CONVERT,
            AConverter.STRING_CONVERT, AConverter.STRING_CONVERT, AConverter.TIMESTAMP_CONVERT};

    private static final boolean[] THROW_ARR = new boolean[]{true, true, true, true, true, true, true};

    @Override
    protected String getRedisKey(Date date) {
        return DataRedisKeyUtils.getKey(Constants.DataRedisKey.TV_ProcessData_YYYYMMDD, date);
    }

    @Override
    public String process(JSONObject target, CacheDataModel cacheData, Date date) {
        return this.list2CSVLine(ConverterFactory.convert(target, FIELD_ARR, CONVERT_ARR, THROW_ARR), SepratorUtil.SEP_PIPE);
    }

    @Override
    public String getDataType() {
        return Constants.PageDataType.TV_RECORD;
    }

    @Override
    public String getTaskType() {
        return Constants.SpiderTaskType.TV_RECORD;
    }
}
